#!/bin/bash
set -e

# 等待数据库启动
echo "等待数据库启动..."
while ! nc -z db 5432; do
  sleep 1
done
echo "数据库已启动"

# 运行数据库迁移
echo "运行数据库迁移..."
python manage.py migrate

# 创建超级用户（如果不存在）
echo "创建超级用户..."
python manage.py shell << EOF
from django.contrib.auth import get_user_model
User = get_user_model()
if not User.objects.filter(username='admin').exists():
    User.objects.create_superuser('admin', 'admin@example.com', 'admin123')
    print('超级用户已创建: admin/admin123')
else:
    print('超级用户已存在')
EOF

# 初始化权限数据
echo "初始化权限数据..."
python manage.py init_permissions

# 启动应用
echo "启动Django应用..."
exec "$@"
